# assistant_pkg/agent.py
from google.adk.agents import Agent
from google.adk.tools import google_search # 尝试导入搜索工具

# 定义我们的信息助手代理
root_agent = Agent(
    name="api_driven_assistant",
    # 使用 Google AI Studio API 支持的模型, 例如 "gemini-1.5-flash-latest" 或 "gemini-pro"
    # 请参考 https://ai.google.dev/models/gemini 获取最新的模型名称
    model="gemini-1.5-flash-latest",
    instruction=(
        "你是一个乐于助人的 AI 信息助手。"
        "请根据你已有的知识清晰、简洁地回答用户的问题。"
        "如果用户的问题涉及到非常近期的事件、实时数据或者你不确定的领域，"
        "你可以尝试使用 'Google Search' 工具来查找信息。"
        "如果使用搜索工具，请基于搜索结果进行回答。"
    ),
    tools=[
        google_search # 添加搜索工具
    ]
)

print(f"代理 '{root_agent.name}' 已定义，使用模型 '{root_agent.model}'。")